DotWriterold.cpp

Go to the documentation of this file.

#include "dotwriter.h"
#include <wx/strconv.h>
#include <wx/file.h> 
#include <wx/msgdlg.h> 
//#include <wx/math.h>


DotWriter::DotWriter()
{
        begin_graph = wxT("digraph\n{");
        end_graph = wxT("}\n");
        fontname = wxT("Arial");
        style = wxT("filled");
        shape = wxT("box");
        cwhite = wxT("white");
        cblack = wxT("black");
        hnode = wxT("");
        dedge = wxT("");
        hedge = wxT("");
        dlabel = wxT("");
        graph = wxT("");
        output = wxT("");
        mlines = NULL;
        m_mgr = NULL;
        dwcn = 0;
        dwce = 0;
        dwtn = 0;
        dwte = 0;
        dwbname = false;
        dwbparam = false;
        writedotfile = false;
}

DotWriter::~DotWriter()
{
        
}

void DotWriter::setLineParser(LineParserList *pLines )//, int numOfLines)
{
        mlines = pLines;
}

void DotWriter::setDotWriterFromDialogSettings(IManager *mgr)
{
        m_mgr = mgr;
        m_mgr->GetConfigTool()->ReadObject(wxT("CallGraph"), &confData);
        dwcn = confData.GetColorsNode();
        dwce = confData.GetColorsEdge();
        dwtn = confData.GetTresholdNode();
        dwte = confData.GetTresholdEdge();
        dwbname = confData.GetBoxName();
        dwbparam = confData.GetBoxParam();
}


void DotWriter::WriteToDotLanguade()
{
        int pl_index = 0;
        float pl_time = 0;
        bool is_node = false;
        wxArrayInt index_pl_nodes;
        
        if (mlines == NULL)
                return;
                
        graph = wxT("graph [ranksep=\"0.25\", fontname=") + fontname + wxT(", nodesep=\"0.125\"];");
        
        hnode = wxT("node [label=\"\\N\", fontname=") + fontname + wxT(", style=") + style + wxT(", height=0, width=0, shape=") + shape + wxT(", fontcolor=") + cwhite + wxT("];");

        hedge = wxT("edge [fontname=") + fontname + wxT("];");

        //graph []; -- not used
        
        output += begin_graph + wxT("\n") + graph + wxT("\n") + hnode + wxT("\n") + hedge + wxT("\n");
        
        LineParserList::compatibility_iterator it = mlines->GetFirst();
        
        while(it)
        {
                LineParser *line = it->GetData();
        
                //if (line->name.IsEmpty()) break;
                
                if(line->pline && wxRound(line->time) >= dwtn) // rozliseni dle procentuelni vytizenosti primarniho uzlu ma vliv na vytvareni uzlu, predat info o pouzitych indexech
                {
                        is_node = true;
                        index_pl_nodes.Add(line->index);
                        //wxMessageBox(wxString::Format(wxT("line->index primary: %i"),line->index));
                        dlabel = wxString::Format(wxT("%i"),line->index); 
                        dlabel += wxT(" [label=\"");
                        dlabel += OptionsShortNameAndParameters(line->name);
                        dlabel += wxT("\\n");
                        dlabel += wxString::Format(wxT("%.2f"), line->time);
                        dlabel += wxT("% \\n");
                        dlabel += wxT("(");
                        dlabel += wxString::Format(wxT("%.2f"), line->self); 
                        dlabel += wxT("%)");
                        dlabel += wxT("\\n");
                        if(line->called0 != -1)
                                dlabel += wxString::Format(wxT("%i"),line->called0) + wxT("x");
                        dlabel += wxT("\",fontcolor=\"");
                        dlabel += DefineColorForLabel(line->time);
                        dlabel += wxT("\", color=\"");
                        dlabel += DefineColorForNodeEdge(line->time, dwcn);
                        //
                        dlabel += wxT("\", fontsize=\"10.00\"];");
                        //
                        output += dlabel + wxT("\n");
                        //
                        dlabel.Clear();
                }
                it = it->GetNext();
        }
        
        it = mlines->GetFirst();
        
        while(it)
        {
                LineParser *line = it->GetData();
                
                //if (line->name.IsEmpty()) break;
                
                if(line->pline)
                {
                        pl_index = line->index; // index for primary node
                        pl_time = line->time; // time for primary node
                }
                
                if (line->child && IsInArray(line->nameid,index_pl_nodes) && IsInArray(pl_index,index_pl_nodes) && (wxRound(pl_time) >= dwte))
                {
                        dedge = wxString::Format(wxT("%i"), pl_index);
                        dedge += wxT(" -> ");
                        dedge += wxString::Format(wxT("%i"),line->nameid);
                        dedge += wxT(" [color=\"");
                        dedge += DefineColorForNodeEdge(pl_time, dwce); // colorarow; //  reseno dle vytizeni hlavniho uzlu
                        dedge += wxT("\", label=\"");
                        //if(line->self != -1)
                        //dedge += wxString::Format(wxT("%.2f"),line->self) + wxT("%");
                        //dedge += wxT("\\n");
                        dedge += wxString::Format(wxT("%i"),line->called0);
                        dedge += wxT("x"); 
                        dedge += wxT("\" ,arrowsize=\"0.50\", fontsize=\"10.00\", fontcolor=\"");
                        dedge += cblack;
                        dedge += wxT("\", labeldistance=\"4.00\", penwidth=\"2.00\"];");                        
                        //
                        output += dedge + wxT("\n");
                        //
                        dedge.Clear();          
                }
                it = it->GetNext();
        }
        output += end_graph;
        if (!is_node)
        {        
                output = wxT("digraph e {0 [label=");
                output += wxT("\"Call Graph is empty, please check settings of node resolutions for this plugin!\"");
                output += wxT(", shape=none, height=2, width=2, fontname=Arial, fontsize=14.00];}");
        }
}

void DotWriter::SendToDotAppOutputDirectory(wxString apppathfolder)
{
        //wxMessageBox(apppathfolder);
        wxString dotfilespath = apppathfolder + stvariables::dotfilesdir + stvariables::sd;
        wxString dottxtpath = dotfilespath + stvariables::dottxtname;
        if(!wxDirExists(dotfilespath))
        {
                wxMkdir(dotfilespath.c_str());
        }
        
        if (wxFileExists(dottxtpath))
                wxRemoveFile(dottxtpath);       
        
        //create new txt file   
        wxFile pFile(dottxtpath, wxFile::write);
        pFile.Open(dottxtpath, wxFile::write);
        
        if(pFile.IsOpened()) 
        {
                writedotfile = pFile.Write(output);
                pFile.Close();
        }
        else
        {
                return;
        }
}

bool DotWriter::DotFileExist(wxString apppathfolder)
{
        wxString dotfilespath = apppathfolder + stvariables::dotfilesdir + stvariables::sd + stvariables::dottxtname;
        
        if (wxFileExists(dotfilespath) && writedotfile) 
                return true;
                else return false;      
}

wxString DotWriter::OptionsShortNameAndParameters(wxString name)
{
        if (name.Contains(wxT('(')) && name.Contains(wxT(')')) && dwbname)
        {
                wxString out = name.BeforeFirst(wxT('('));
                out += wxT("()"); // for function return just ()
                return out;
        
        }
        else if (name.Contains(wxT('(')) && name.Contains(wxT(')')) && dwbparam)
        {
                wxString out = name.BeforeFirst(wxT('('));
                wxString sub = name.AfterFirst(wxT('(')).BeforeFirst(wxT(')'));
                if (sub.IsEmpty())
                {
                        out += wxT("\\n(\\n)");
                        return out;
                }
                else if(sub.Contains(wxT(",")))
                {
                        sub.Replace(wxT(","), wxT(",\\n"));
                        out += wxT("\\n(\\n") + sub + wxT("\\n)");
                        return out;
                }
                else
                {
                        out += wxT("\\n(\\n") + sub + wxT("\\n)");
                        return out;
                
                }
        }
        else return name;
}

wxString DotWriter::DefineColorForNodeEdge(float time, int dwc)
{
        wxString colors[11] = {wxT("#006837"), wxT("#1a9850"), wxT("#66bd63"), wxT("#a6d96a"), wxT("#d9ef8b"), wxT("#fee08b"), wxT("#fdae61"), wxT("#f46d43"), wxT("#d73027") ,wxT("#a50026"), wxT("a50026")};
        int index = 0;
        int interval = 100 / (dwc-1);
        /*
        if (time == 100.00)
        {
                index = (time - 2) / interval;
        }
        else 
        {
                index = time / interval;
        }
        
        if (dwc > 1)
        {
                        return colors[index];
        }
        else return colors[0];
        */
        if(dwc == 1)
        {
                index = 0;
        }
        else if (time == 0.00)
        {
                index = 0;
        }
        else if (time == 100.00 && dwc == 2)
        {
                index = 1;
        }
        else
        {
                index = time / interval;
                wxMessageBox(wxString::Format(wxT("index: %i"),index));
        }
        
        return colors[index];
}

bool DotWriter::IsInArray(int index, wxArrayInt arr)
{
        for(unsigned int i = 0; i < arr.GetCount(); i++)
        {
                if (arr.Item(i) == index) return true;
        }
        return false;

}

wxString DotWriter::DefineColorForLabel(float time)
{
        int index = 0;
        int interval = 100 / dwcn;
        
        if (time == 100.00)
        {
                index = (time - 2) / interval;
        }
        else 
        {
                index = time / interval;
        }
        
        if ((index < 3) || (index > 6))
        {
                return cwhite;
        }
        else
        {
                return cblack;
        }
}